package com.sxgo.sxgoserver.base.file;

import com.sxgo.sxgoserver.base.response.JsonResult;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.UUID;

@Slf4j
@RestController
@Api(tags = "图片上传模块")
@RequestMapping("/v1/file")
public class UploadController {
    /**
     * 定义图片上传的方法
     * MultopartFile 用来接收前端上传的图片文件对象
     *
     * @param file 注意:此名字必须与upload上传组件的name名一致
     * @return
     */
    @PostMapping("/upload")
    @ApiOperation(value = "上传功能")
    public JsonResult upload(MultipartFile file) {
        String fileName = file.getOriginalFilename();
        //2.获取图片后缀名
        String suffix = fileName.substring(fileName.lastIndexOf("."));
        //3.生成一个随机串作为新的图片名
        fileName = UUID.randomUUID() + suffix;
        //4.指定一个磁盘路径,作为保存图片文件的根路径
        String dirPath = "d:/files/img";
        //5.为了提高检索性能,我们可以设置一颗日期格式生成对应的文件夹类存储图片文件
        //这里采用的是年月日的格式
        SimpleDateFormat sdf = new SimpleDateFormat("/yyyy/MM/dd/");
        //6.将当前真实的日期转为上面的格式
        String datePath = sdf.format(new Date());
        //7.将上面的磁盘路径与日期路径拼接成一个完整的图片文件存放路径
        File dirFile = new File(dirPath + datePath);
        if (!dirFile.exists()) {
            dirFile.mkdirs();
        }
        //8.拼接图片的完整路径
        String filePath = dirPath + datePath + fileName;
        //9.将图片文件保存到上面路径下
        try {
            file.transferTo(new File(filePath));
            return JsonResult.ok("/img"+datePath + fileName);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    @ApiOperation(value = "移除功能")
    @PostMapping("remove")
    public JsonResult remove(String imgUrl) {
        log.debug("删除图片 imgUrl:{}", imgUrl);
        new File("d:/files" + imgUrl).delete();
        return JsonResult.ok();
    }
}
